package com.liang.leetcode.list;

public class List01_RemoveElements {
    public static void main(String[] args) {
        // 创建一个数组作为链表的元素
        int[] arr = {1, 2, 6, 3, 4, 5, 6};

        // 创建链表
        ListNode head = ListNodeUtil.createLinkedList(arr);
        ListNode list = removeElements(head, 6);

        // 打印链表
        ListNodeUtil.printLinkedList(list);
    }

    /**
     * 移除链表中指定值的节点
     * @param head 链表头节点
     * @param val  指定值
     * @return 处理后的链表头节点
     */
    public static ListNode removeElements(ListNode head, int val) {
        // 删除头节点，直到头节点不等于目标值或链表为空
        while (head != null && head.val == val) {
            head = head.next;
        }

        // 删除非头结点
        ListNode cur = head;
        // 遍历链表
        while (cur != null && cur.next != null) {
            // 如果当前节点的下一个节点的值等于目标值
            if (cur.next.val == val) {
                // 删除当前节点的下一个节点
                cur.next = cur.next.next;
            } else
                // 否则，移动当前指针到下一个节点
                cur = cur.next;
        }
        // 返回处理后的链表头节点
        return head;
    }

    /**
     * 使用虚拟头节点，移除链表中指定值的节点
     */
    public static ListNode removeElements2(ListNode head, int val) {
        // 如果链表为空，直接返回 null
        if (head == null) {
            return null;
        }
        // 创建虚拟头节点，简化链表操作
        ListNode dummyHead = new ListNode(-1, head);
        //定义临时指针指向虚拟头节点
        ListNode cur = dummyHead;
        while (cur.next != null) {
            // 如果当前节点的下一个节点的值等于目标值
            if (cur.next.val == val) {
                // 删除当前节点的下一个节点
                cur.next = cur.next.next;
            } else
                // 否则，移动当前指针到下一个节点
                cur = cur.next;
        }
        // 返回虚拟头节点的下一个节点，即处理后的链表头部
        return dummyHead.next;
    }


}

